module Qwen
  class << self
    def open_ai_stock(code)
      dashscope_api_key = "sk-2866f5c164e04e409b6d65587e887d53"
      connection = Faraday.new(url: 'https://dashscope.aliyuncs.com') do |faraday|
        faraday.options.timeout = 60
        faraday.headers['Authorization'] = "Bearer #{dashscope_api_key}"
        faraday.headers['Content-Type'] = "application/json"
      end
      body = {
        # model: 'qwen-plus',
        # model: 'qwen-max-latest',
        model: 'qwen-plus-latest',
        # model: 'deepseek-r1',
        messages: [
          {
            role: 'system',
            content: '金融分析师、股票市场调研专家、高级基金经理、股神、数据分析师、数据挖掘专家
                      要求：（重点：绝对不许胡说，乱说，一定、必须、务必要说准确的信息）
                      1、给出具体的数字，不要用xx来表示
                      2、尽量使用2025年的数据
                      3、使用严格标准markdown格式输出(rails gem redcarpet渲染markdown格式的table)
                      如果输入的是一只股票的名字 或者 股票代码：
                      输出内容：
                      1、股票的基本面：(包括最近一个季度每股收益、主营业务、行业地位等）
                      2、股票关联哪些行业
                      3、股票有哪些热点概念
                      4、营收、利润、增长情况以及风险点
                      5、主营生产的产品有哪些品牌，哪些具体的产品？
                      6、销售额最高的10款产品
                      7、交易风险提示(包括但不限于：股东减持、业绩预告、业绩快报、业绩预增、业绩预减、业绩大幅下滑、业绩大幅上升、股东增持、股东回购、分红派息等)
                      8、股票的最新消息(5条消息，包含消息来源、时间)'
          },
          {
            role: 'user',
            content: code
          }
        ],
      }
      response = connection.post do |req|
        req.url '/compatible-mode/v1/chat/completions'
        # req.url 'compatible-mode/v1'
        req.body = body.to_json
      end
      JSON.parse(response.body)["choices"][0]["message"]["content"]
    end
  end
end
